import torch
import my_ops

def test_add():
    a = torch.rand([10, 10, 3])
    b = torch.rand([10, 10, 3])
    c = my_ops.my_add(a, b)
    d = a + b
    assert torch.allclose(c, d)
test_add()